const {promisify} = require('es6-promisify');
export default (db = {}) => {
  const funs = [
    'insert',
    'update',
    'remove',
    'find',
    'findOne',
    'count',
    // 'skip',
    // 'limit',
    // 'exec',
  ];

  const funMap = [
    ['insert', ['insertAsync']],
    ['update', ['updateAsync']],
    ['remove', ['removeAsync']],
    ['find', ['findAsync']],
    ['findOne', ['findOneAsync']],
    ['count', ['countAsync']],
  ];
  for (const [oldKey, newKey] of funMap) {
    db[newKey] = promisify(db[oldKey]);
  }
};
